package com.example.config;


import javax.sql.DataSource;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.PropertySource;

import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.support.http.StatViewServlet;
import com.alibaba.druid.support.http.WebStatFilter;

@Configuration
@PropertySource(value="classpath:druid.properties")
public class DruidConfiguration {
		@Value("${spring.datasource.url}")
		private String url;
		@Value("${spring.datasource.username}")
		private String username;
		@Value("${spring.datasource.password}")
		private String password;
		@Value("${spring.datasource.driver-class-name}")
		private String driverClassName;
	
	    @Bean(destroyMethod = "close", initMethod = "init")
	    @ConfigurationProperties(prefix = "spring.datasource")
	    public DataSource druidDataSource() {
	        DruidDataSource druidDataSource = new DruidDataSource();
	        druidDataSource.setUrl(url);
	        druidDataSource.setUsername(username);
	        druidDataSource.setPassword(password);
	        druidDataSource.setDriverClassName(driverClassName);
	        return druidDataSource;
	    }

	    /**
	     * 注册一个StatViewServlet
	     * @return
	     */
	    @Bean
	    public ServletRegistrationBean druidStatViewServlet(){
	        //org.springframework.boot.context.embedded.ServletRegistrationBean提供类的进行注册.
	        ServletRegistrationBean servletRegistrationBean = new ServletRegistrationBean(new StatViewServlet(),"/druid/*");

	        //添加初始化参数：initParams
	        //白名单：
	        servletRegistrationBean.addInitParameter("allow","127.0.0.1,116.205.13.8");
	        //IP黑名单 (存在共同时，deny优先于allow) : 如果满足deny的话提示:Sorry, you are not permitted to view this page.
	        servletRegistrationBean.addInitParameter("deny","192.168.1.73");
	        //登录查看信息的账号密码.
	        servletRegistrationBean.addInitParameter("loginUsername","admin");
	        servletRegistrationBean.addInitParameter("loginPassword","123456");
	        //是否能够重置数据.
	        servletRegistrationBean.addInitParameter("resetEnable","false");
	        return servletRegistrationBean;
	    }

	    /**
	     * 注册一个：filterRegistrationBean
	     * @return
	     */
	    @Bean
	    public FilterRegistrationBean druidStatFilter(){

	        FilterRegistrationBean filterRegistrationBean = new FilterRegistrationBean(new WebStatFilter());

	        //添加过滤规则.
	        filterRegistrationBean.addUrlPatterns("/*");

	        //添加不需要忽略的格式信息.
	        filterRegistrationBean.addInitParameter("exclusions","*.js,*.gif,*.jpg,*.png,*.css,*.ico,/druid/*");
	        return filterRegistrationBean;
	    }
	
}
